tensor([[0.4600, 0.4600, 0.4600, 0.4600, 0.4600],
        [0.5400, 0.5400, 0.5400, 0.5400, 0.5400]], grad_fn=<RoundBackward1>)